""" Installing AutoGluon
conda create -n myenv python=3.9 cudatoolkit=11.3 -y
conda activate myenv

pip install -U pip
pip install -U setuptools wheel

#Install the proper version of PyTorch following https://pytorch.org/get-started/locally/
pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 --extra-index-url https://download.pytorch.org/whl/cu116

python check.py

pip install autogluon
"""

import torch
print(torch.cuda.is_available())  # Should be True
print(torch.cuda.device_count())  # Should be > 0